# _*_ encoding:utf-8 _*_

"""
    自定义一个判断用户登录的拓展视图类
"""

from django.contrib.auth.mixins import LoginRequiredMixin
from django.http import JsonResponse


class LoginRequiredJSONMixin(LoginRequiredMixin):

    def handle_no_permission(self):
        # 默认该函数返回的是一个重定向响应对象HttpResponseRedirect
        # 为了和工程一直，需要返回一个JsonResponse
        return JsonResponse({'code': 400, 'errmsg': '用户未登录'}, status=400, json_dumps_params={'ensure_ascii': False})


